Home
last modified time | relevance | path

Searched refs:packed_var (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dpacked_distributed_variable_test.py45 packed_var = packed_distributed_variable.PackedDistributedVariable([v0, v1])
46 self.assertFalse(packed_var.handle.is_packed)
47 self.assertTrue(packed_var.is_initialized)
50 self.assertAllEqual(packed_var.get_var_on_current_device(), v0)
51 val0 = packed_var.assign(2.0).assign_add(1.0)
55 self.assertAllEqual(packed_var.get_var_on_current_device(), v1)
56 val0 = packed_var.assign(2.0).assign_add(1.0)
61 self.assertTrue(packed_var.handle.is_packed)
63 packed_var.assign_add(3.0).assign_sub(1.0)
64 read0 = packed_var.value()
[all …]
Dvalues_util.py83 packed_var = var._packed_variable # pylint: disable=protected-access
84 if packed_var is not None:
87 assign_on_device(d, packed_var, tensor)
88 for d in packed_var.devices))
Dtpu_strategy.py1395 packed_var = var._packed_variable # pylint: disable=protected-access
1396 if packed_var is not None and not context.executing_eagerly():
1398 return fn(packed_var, *args, **kwargs)
1400 return (fn(packed_var, *args, **kwargs),)
1406 if packed_var is not None:
1407 for device in packed_var.devices:
1408 values_and_devices.append((packed_var, device))
/external/mesa3d/src/compiler/glsl/
Dlower_packed_varyings.cpp617 ir_variable *packed_var = packed_deref->variable_referenced(); in lower_rvalue() local
619 packed_var->data.stream |= in lower_rvalue()
715 ir_variable *packed_var = new(this->mem_ctx) in get_packed_varying_deref() local
721 packed_var->data.max_array_access = this->gs_input_vertices - 1; in get_packed_varying_deref()
723 packed_var->data.centroid = unpacked_var->data.centroid; in get_packed_varying_deref()
724 packed_var->data.sample = unpacked_var->data.sample; in get_packed_varying_deref()
725 packed_var->data.patch = unpacked_var->data.patch; in get_packed_varying_deref()
726 packed_var->data.interpolation = in get_packed_varying_deref()
729 packed_var->data.location = location; in get_packed_varying_deref()
730 packed_var->data.precision = unpacked_var->data.precision; in get_packed_varying_deref()
[all …]
/external/tensorflow/tensorflow/python/eager/
Dremote_test.py465 packed_var = ops.pack_eager_tensors([var0.handle, var1.handle])
466 self.assertEqual(packed_var.device,
468 self.assertEqual(packed_var.backing_device,
475 packed_var, dtype=dtypes.float32)
478 packed_var, dtype=dtypes.float32)
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_sample_aos.c1130 LLVMValueRef packed_var, packed; in lp_build_sample_aos() local
1149 packed_var = lp_build_alloca(bld->gallivm, u8n_bld.vec_type, "packed_var"); in lp_build_sample_aos()
1157 packed_var); in lp_build_sample_aos()
1184 packed_var); in lp_build_sample_aos()
1193 packed_var); in lp_build_sample_aos()
1198 packed = LLVMBuildLoad(builder, packed_var, ""); in lp_build_sample_aos()
/external/mesa3d/src/compiler/nir/
Dnir_serialize.c204 union packed_var { union
236 STATIC_ASSERT(sizeof(union packed_var) == 4); in write_variable()
237 union packed_var flags; in write_variable()
342 union packed_var flags; in read_variable()